speed/docs/config |
您所在的位置:网站首页 › speed config › speed/docs/config |
PyTorch DeepSpeed Config JSON Documentation
REQUIRED DeepSpeed Config JSON Parameters
train_batch_size: [integer] Value Example The effective training batch size. This is the amount of data samples that leads to one step of model update. train_batch_size is aggregated by the batch size that a single GPU processes in one forward/backward pass (a.k.a., train_step_batch_size), the gradient accumulation steps (a.k.a., gradient_accumulation_steps), and the number of GPUs. 32 OPTIONAL DeepSpeed Config JSON Parameters Batch Size Related Parameterstrain_micro_batch_size_per_gpu: [integer] Description Default Batch size to be processed by one GPU in one step (without gradient accumulation). When specified, gradient_accumulation_steps is automatically calculated using train_batch_size and number of GPUs. Should not be concurrently specified with gradient_accumulation_steps in the configuration JSON. train_batch_size valuegradient_accumulation_steps: [integer] Description Default Number of training steps to accumulate gradients before averaging and applying them. This feature is sometimes useful to improve scalability since it results in less frequent communication of gradients between steps. Another impact of this feature is the ability to train with larger batch sizes per GPU. When specified, train_step_batch_size is automatically calculated using train_batch_size and number of GPUs. Should not be concurrently specified with train_step_batch_size in the configuration JSON. 1 Optimizer Parametersoptimizer: [dictionary] Fields Value Example type The optimizer name. DeepSpeed natively supports Adam and LAMB optimizers and will import other optimizers from torch. "Adam" params Dictionary of parameters to instantiate optimizer. The parameter names must match the optimizer constructor signature (e.g., for Adam). {"lr": 0.001, "eps": 1e-8}Example of optimizer "optimizer": { "type": "Adam", "params": { "lr": 0.001, "betas": [ 0.8, 0.999 ], "eps": 1e-8, "weight_decay": 3e-7 } } Scheduler Parametersscheduler: [dictionary] Fields Value Example type The scheduler name. See here for list of support schedulers. "1Cycle" params Dictionary of parameters to instantiate scheduler. The parameter names should match scheduler constructor signature. {"lr": 0.001, "eps": 1e-8}Example of scheduler "scheduler": { "type": "WarmupLR", "params": { "warmup_min_lr": 0, "warmup_max_lr": 0.001, "warmup_num_steps": 1000 } } Communication optionsfp32_allreduce: [boolean] Description Default During gradient averaging perform allreduce with 32 bit values falsedisable_allgather: [boolean] Description Default Disable allgather when using ZeRO optimizer and instead use broadcast falseprescale_gradients: [boolean] Description Default Scale gradients before doing allreduce falsesparse_gradients: [boolean] Description Default Enable sparse compression of torch.nn.Embedding gradients. false FP16 training optionszero_optimization: [boolean] Description Default Enable ZeRO memory optimization wrapper for FP16 Training. Currently compatible only with Adam optimizer. falsefp16: [dictionary] Description Default Configuration for using mixed precision/FP16 training that leverages NVIDIA's Apex package. An example, including the available dictionary keys is illustrated below. None "fp16": { "enabled": true, "loss_scale": 0, "initial_scale_power": 32, "loss_scale_window": 1000, "hysteresis": 2, "min_loss_scale": 1 }fp16:enabled: [boolean] Description Default enabled is a fp16 parameter indicating whether or not FP16 training enabled. falsefp16:loss_scale: [float] Description Default loss_scale is a fp16 parameter representing the loss scaling value for FP16 training. The default value of 0.0 results in dynamic loss scaling, otherwise the value will be used for static fixed loss scaling. 0.0fp16:initial_scale_power: [integer] Description Default initial_loss_scale_power is a fp16 parameter representing the power of the initial dynamic loss scale value. The actual loss scale is computed as 2initial_loss_scale_power. 32fp16:loss_scale_window: [integer] Description Default loss_scale_window is a fp16 parameter representing the window over which to raise/lower the dynamic loss scale value. 1000fp16:hysteresis: [integer] Description Default hysteresis is a fp16 parameter representing the delay shift in dynamic loss scaling. 2fp16:min_loss_scale: [integer] Description Default min_loss_scale is a fp16 parameter representing the minimum dynamic loss scale value. 1000 Gradient Clippinggradient_clipping: [float] Description Default Enable gradient clipping with value 0 Loggingsteps_per_print: [integer] Description Default Print train loss every N steps 10wall_clock_breakdown: [boolean] Description Default Enable timing of the latency of forward/backward/update training phases falsedump_state: [boolean] Description Default Print out state information of DeepSpeed object after initialization false |
今日新闻 |
点击排行 |
|
推荐新闻 |
图片新闻 |
|
专题文章 |
CopyRight 2018-2019 实验室设备网 版权所有 win10的实时保护怎么永久关闭 |